python -u main.py --batch_size=8 --step_size=100 --total_itrs=1000
